package com.qiyee.dalligent.common.utils;

import com.qiyee.dalligent.common.execption.BaseException;
import lombok.extern.slf4j.Slf4j;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;

/**
 * @author 石亚宁 on 2021-09-03 13:41:48
 * @ClassName : com.qiyee.job.common.utils.FileDownloadUtils
 * @Description : 类描述
 */
@Slf4j
public class FileDownloadUtils {
    public static HttpServletResponse download(String path, HttpServletResponse response) {
        OutputStream toClient=null;
        try {
            // path是指欲下载的文件的路径。
            File file = new File(path);
            // 取得文件名。
            String filename = file.getName();
            // 取得文件的后缀名。
            String ext = filename.substring(filename.lastIndexOf(".") + 1).toUpperCase();

            // 以流的形式下载文件。
            InputStream fis = new BufferedInputStream(new FileInputStream(path));
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            // 清空response
            response.reset();
            // 设置response的Header
            response.addHeader("Content-Disposition", "attachment;filename=" + new String(filename.getBytes()));
            response.addHeader("Content-Length", "" + file.length());
            toClient = new BufferedOutputStream(response.getOutputStream());
            response.setContentType("application/octet-stream");
            toClient.write(buffer);
            toClient.flush();
            deleteFile(path);

        } catch (IOException ex) {
            ex.printStackTrace();
        }finally {
            try {
                toClient.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return response;
    }

    public static void downloadLocal(HttpServletResponse response, String fileName, String downloadFile) throws FileNotFoundException {
        // 下载本地文件

        // 读到流中
        InputStream inStream = new FileInputStream(downloadFile);// 文件的存放路径
        // 设置输出的格式
        response.reset();
        response.setContentType("bin");
        response.addHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
        // 循环取出流中的数据
        byte[] b = new byte[1024];
        int len;
        try {
            while ((len = inStream.read(b)) > 0){
                response.getOutputStream().write(b, 0, len);
            }

            inStream.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void downloadNet(HttpServletResponse response) throws MalformedURLException {
        // 下载网络文件
        int bytesum = 0;
        int byteread = 0;

        URL url = new URL("windine.blogdriver.com/logo.gif");

        try {
            URLConnection conn = url.openConnection();
            InputStream inStream = conn.getInputStream();
            FileOutputStream fs = new FileOutputStream("c:/abc.gif");

            byte[] buffer = new byte[1204];
            int length;
            while ((byteread = inStream.read(buffer)) != -1) {
                bytesum += byteread;
                System.out.println(bytesum);
                fs.write(buffer, 0, byteread);
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 删除文件
     *
     * @param filePath 文件
     * @return
     */
    public static boolean deleteFile(String filePath) {
        boolean flag = false;
        File file = new File(filePath);
        // 路径为文件且不为空则进行删除
        if (file.isFile() && file.exists()) {
            file.delete();
            flag = true;
            log.info("删除成功："+filePath);
        }
        return flag;
    }
    public static String getTempDir(){

        String dir = System.getProperty("java.io.tmpdir") + File.separator;
        File fDir = new File(dir);
        if (!fDir.exists()) {
            fDir.mkdirs();
        }
        log.info("文件路径："+fDir.getPath());
        return fDir.getPath();
    }

    public  static void downLoadResume(HttpServletResponse response,String fileName){
        String filePath="";
        if (OSinfo.isWindows()){
            filePath=UploadUtil.WINDOWS_FILE_PATH+fileName;
        }
        if (OSinfo.isLinux()){
            filePath=UploadUtil.LINUX_FILE_PATH+fileName;
        }
        try {
            downloadLocal(response,fileName,filePath);
        } catch (FileNotFoundException e) {
            log.error("下载失败");
            throw new BaseException("下载失败,没找到文件");
        }
    }
}
